<template>
  <div class="flow-wraper">
    <VueFlow :nodes="nodes" :edges="edges" fit-view-on-init>
      <template #edge-button="buttonEdgeProps">
        <EdgeWithButton
          :id="buttonEdgeProps.id"
          :source-x="buttonEdgeProps.sourceX"
          :source-y="buttonEdgeProps.sourceY"
          :target-x="buttonEdgeProps.targetX"
          :target-y="buttonEdgeProps.targetY"
          :source-position="buttonEdgeProps.sourcePosition"
          :target-position="buttonEdgeProps.targetPosition"
          :marker-end="buttonEdgeProps.markerEnd"
          :style="buttonEdgeProps.style"
        />
      </template>

      <template #edge-custom="customEdgeProps">
        <CustomEdge
          :id="customEdgeProps.id"
          :source-x="customEdgeProps.sourceX"
          :source-y="customEdgeProps.sourceY"
          :target-x="customEdgeProps.targetX"
          :target-y="customEdgeProps.targetY"
          :source-position="customEdgeProps.sourcePosition"
          :target-position="customEdgeProps.targetPosition"
          :data="customEdgeProps.data"
          :marker-end="customEdgeProps.markerEnd"
          :style="customEdgeProps.style"
        />
      </template>

      <Background />
    </VueFlow>
  </div>
</template>

<script setup>
import { h, ref } from 'vue'
import { Background } from '@vue-flow/background'
import { MarkerType, VueFlow } from '@vue-flow/core'
import EdgeWithButton from './EdgeWithButton.vue'
import CustomEdge from './CustomEdge.vue'
import CustomEdgeLabel from './CustomEdgeLabel.vue'

const nodes = ref([
  {
    id: '1',
    type: 'input',
    label: 'Start',
    position: { x: 50, y: 0 },
    style: { borderColor: '#10b981' },
  },
  { id: '2', label: 'Node 2', position: { x: 150, y: 100 } },
  { id: '2a', label: 'Node 2a', position: { x: 0, y: 180 } },
  { id: '3', label: 'Node 3', position: { x: 250, y: 200 } },
  { id: '4', label: 'Node 4', position: { x: 400, y: 300 } },
  { id: '3a', label: 'Node 3a', position: { x: 175, y: 300 } },
  { id: '5', label: 'Node 5', position: { x: 200, y: 400 } },
  { id: '6', type: 'output', label: 'Output 6', position: { x: 0, y: 350 } },
  { id: '7', type: 'output', label: 'Output 7', position: { x: 50, y: 600 } },
  { id: '8', type: 'output', label: 'Output 8', position: { x: 350, y: 600 } },
  { id: '9', type: 'output', label: 'Output 9', position: { x: 550, y: 400 } },
])

const edges = ref([
  { id: 'e1-2', source: '1', target: '2', label: 'bezier edge (default)', class: 'normal-edge' },
  { id: 'e2-2a', source: '2', target: '2a', type: 'smoothstep', label: 'smoothstep edge' },
  { id: 'e2-3', source: '2', target: '3', type: 'step', label: 'step edge' },
  { id: 'e3-4', source: '3', target: '4', type: 'straight', label: 'straight edge' },
  {
    id: 'e3-3a',
    source: '3',
    target: '3a',
    type: 'straight',
    label: 'label only edge',
    style: { stroke: 'none' },
  },
  {
    id: 'e3-5',
    source: '4',
    target: '5',
    animated: true,
    label: 'animated styled edge',
    style: { stroke: '#10b981' },
  },
  {
    id: 'e2a-6',
    source: '2a',
    target: '6',
    label: () => h(CustomEdgeLabel, { label: 'custom label text' }),
    labelStyle: { fill: '#10b981', fontWeight: 700 },
    markerEnd: MarkerType.Arrow,
  },
  {
    id: 'e5-7',
    source: '5',
    target: '7',
    label: 'label with bg',
    labelBgPadding: [8, 4],
    labelBgBorderRadius: 4,
    labelBgStyle: { fill: '#FFCC00', color: '#fff', fillOpacity: 0.7 },
    markerEnd: MarkerType.ArrowClosed,
  },
  {
    id: 'e5-8',
    source: '5',
    target: '8',
    type: 'button',
    data: { text: 'custom edge' },
    markerEnd: MarkerType.ArrowClosed,
  },
  {
    id: 'e4-9',
    source: '4',
    target: '9',
    type: 'custom',
    data: { text: 'styled custom edge label' },
  },
])
</script>

<style>
@import './styles.css';
.flow-wraper {
  position: relative;
  height: 100%;
  .controls {
    position: absolute;
    z-index: 1;
  }
}
</style>
